/*
 * Copyright 2011 www.pretty-tools.com
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */


#pragma once
#include "resource.h"       // main symbols

#include "XPCOMAdapter_i.h"
#include <atlctl.h>

#if defined(_WIN32_WCE) && !defined(_CE_DCOM) && !defined(_CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA)
#error "Single-threaded COM objects are not properly supported on Windows CE platform, such as the Windows Mobile platforms that do not include full DCOM support. Define _CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA to force ATL to support creating single-thread COM object's and allow use of it's single-threaded COM object implementations. The threading model in your rgs file was set to 'Free' as that is the only threading model supported in non DCOM Windows CE platforms."
#endif

// CWebBrowserPersist

class ATL_NO_VTABLE CWebBrowserPersist
    : public CComObjectRootEx<CComSingleThreadModel>
	, public CComCoClass<CWebBrowserPersist, &CLSID_WebBrowserPersist>
	, public IDispatchImpl<IWebBrowserPersist, &IID_IWebBrowserPersist, &LIBID_XPCOMAdapterLib, /*wMajor =*/ 1, /*wMinor =*/ 0>
	, public IObjectSafetyImpl<CWebBrowserPersist, INTERFACESAFE_FOR_UNTRUSTED_CALLER>
{
public:
	CWebBrowserPersist();
	~CWebBrowserPersist();

DECLARE_REGISTRY_RESOURCEID(IDR_WEBBROWSERPERSIST)

BEGIN_CATEGORY_MAP(CWebBrowserPersist)
	IMPLEMENTED_CATEGORY(CATID_SafeForScripting)
	IMPLEMENTED_CATEGORY(CATID_SafeForInitializing)
END_CATEGORY_MAP()

BEGIN_COM_MAP(CWebBrowserPersist)
	COM_INTERFACE_ENTRY(IDispatch)
	COM_INTERFACE_ENTRY(IWebBrowserPersist)
	COM_INTERFACE_ENTRY(IObjectSafety)
END_COM_MAP()

	DECLARE_PROTECT_FINAL_CONSTRUCT()

	HRESULT FinalConstruct()
	{
		return S_OK;
	}

	void FinalRelease()
	{
	}

protected:
    class BindStatusCallback : public IBindStatusCallback, public CComObjectRootEx<CComSingleThreadModel>
    {
    public:
        CComPtr<IDispatch> m_pProgressListener;

    public:
        BEGIN_COM_MAP(BindStatusCallback)
            COM_INTERFACE_ENTRY(IBindStatusCallback)
        END_COM_MAP()

        DECLARE_NOT_AGGREGATABLE(BindStatusCallback) 

        virtual HRESULT STDMETHODCALLTYPE OnStartBinding( 
            /* [in] */ DWORD dwReserved,
            /* [in] */ __RPC__in_opt IBinding *pib)
        {
            ATLTRACE("OnStartBinding(...)\n");
            return S_OK;
        }
        
        virtual HRESULT STDMETHODCALLTYPE GetPriority( 
            /* [out] */ __RPC__out LONG *pnPriority)
        {
            ATLTRACE("GetPriority(...)\n");
            *pnPriority = THREAD_PRIORITY_NORMAL;
            return S_OK;
        }
        
        virtual HRESULT STDMETHODCALLTYPE OnLowResource( 
            /* [in] */ DWORD reserved)
        {
            ATLTRACE("OnLowResource(...)\n");
            return S_OK;
        }
        
        virtual HRESULT STDMETHODCALLTYPE OnStopBinding( 
            /* [in] */ HRESULT hresult,
            /* [unique][in] */ __RPC__in_opt LPCWSTR szError);
        
        virtual /* [local] */ HRESULT STDMETHODCALLTYPE GetBindInfo( 
            /* [out] */ DWORD *grfBINDF,
            /* [unique][out][in] */ BINDINFO *pbindinfo)
        {
            ATLTRACE("GetBindInfo(...)\n");
            *grfBINDF = BINDF_ASYNCHRONOUS;
            return S_OK;
        }
        
        virtual /* [local] */ HRESULT STDMETHODCALLTYPE OnDataAvailable( 
            /* [in] */ DWORD grfBSCF,
            /* [in] */ DWORD dwSize,
            /* [in] */ FORMATETC *pformatetc,
            /* [in] */ STGMEDIUM *pstgmed)
        {
            ATLTRACE("OnDataAvailable(...)\n");
            return S_OK;
        }
        
        virtual HRESULT STDMETHODCALLTYPE OnObjectAvailable( 
            /* [in] */ __RPC__in REFIID riid,
            /* [iid_is][in] */ __RPC__in_opt IUnknown *punk)
        {
            ATLTRACE("OnObjectAvailable(...)\n");
            return S_OK;
        }

        STDMETHOD(OnProgress)( 
            /* [in] */ ULONG ulProgress,
            /* [in] */ ULONG ulProgressMax,
            /* [in] */ ULONG ulStatusCode,
            /* [unique][in] */ __RPC__in_opt LPCWSTR szStatusText);
    };

protected:
    CComPtr<IDispatch> m_pProgressListener;

public:
    STDMETHOD(get_progressListener)(IDispatch** pVal);
    STDMETHOD(put_progressListener)(IDispatch* newVal);
    STDMETHOD(saveURI)(BSTR uri, ILocalFile* destFile);
};

OBJECT_ENTRY_AUTO(__uuidof(WebBrowserPersist), CWebBrowserPersist)
